GEM: A Generative Embedding Model Bridging Reasoning and Retrieval
GEM:桥接推理与检索的生成式嵌入模型
🧭 生成式嵌入 · 推理增强检索 ★ 深度解读
University of Glasgow
GEM 用单一生成式嵌入模型先对 query 推理再编码,使检索模型能理解推理而不仅是表面匹配,4B 模型超越更大 baseline
📌 工作参考价值(4 点)
- 在搜索 query 理解阶段,可让 retriever 自身生成 query 解析(意图、相关标准、负向条件),再对 query+推理拼接后编码,而非仅依赖原始 query 或外部 query expansion。电商搜索中复杂长尾 query(例如“适合送长辈的安卓手机,不要小米”)可先用 LLM 生成明确检索标准,提升召回精度。
- 训练 embedding 模型时,保留生成能力(联合 causal LM + contrastive loss),用专门 <|embed|> token 而非 EOS token,推理时可复用 KV cache,避免先推理再编码的额外计算;业务部署时可在同一模型内完成 query 推理和向量化,降低 pipeline 延迟。
- 数据合成 trick:采样多个 reasoning 候选,用 LLM 判断原始相关文档是否仍满足该推理来过滤;再基于通过过滤的推理生成正文档和“主题相似但违背某个推理标准”的 hard negatives。这种对齐数据能显著提升 embedding 对指令/约束的敏感度,适用于电商推荐中“可解释偏好”或“多条件过滤”场景。
- 测试期 compute scaling:通过 prompt 要求生成更长推理(如目标词数)可稳定提升 nDCG,但到 1024 词后饱和;工程上可对高价值 query 分层使用,长尾复杂 query 用扩展推理,普通 query 短推理,平衡效果与成本。